package org.example.leetCode;

import org.example.entity.TreeNode;

/**
 * 700. 二叉搜索树中的搜索
 * 给定二叉搜索树（BST）的根节点 root 和一个整数值 val。
 * <p>
 * 你需要在 BST 中找到节点值等于 val 的节点。 返回以该节点为根的子树。 如果节点不存在，则返回 null 。
 */
public class Code700 {
    public TreeNode searchBST(TreeNode root, int val) {
        while (root != null) {
            if (val == root.val) {
                return root;
            }
            root = root.val > val ? root.left : root.right;
        }
        return null;
    }

    public static void main(String[] args) {
        // 创建一个测试用的二叉搜索树
        //       4
        //      / \
        //     2   7
        //    / \
        //   1   3

        TreeNode root = new TreeNode(4);
        root.left = new TreeNode(2);
        root.right = new TreeNode(7);
        root.left.left = new TreeNode(1);
        root.left.right = new TreeNode(3);

        Code700 solution = new Code700();

        // 测试查找存在的节点
        TreeNode result1 = solution.searchBST(root, 2);
        System.out.println("查找节点2的结果: " + (result1 != null ? result1.val : "未找到"));

        // 测试查找不存在的节点
        TreeNode result2 = solution.searchBST(root, 5);
        System.out.println("查找节点5的结果: " + (result2 != null ? result2.val : "未找到"));
    }
}
